
from stu_db import init_table,Stu 

#初始化一个stubase数据库和stu表
init_table() 

#创建学生信息表对象
stu = Stu() 

def ShowStu(stu):
	
	#学生信息的输出函数
	
	stulist = stu.findAll()
	if len(stulist) == 0:
		print('==========没有学员信息可以输出！=========')
	else:
		print("|{0:<5}| {1:<10}| {2:<5}| {3:<10}|".format("sid","name","age","classid"))
		print("-"*40)
		for record in stulist:
			print("|{0:<5}| {1:<10}| {2:<5}| {3:<10}|".format(record[0],record[1],record[2],record[3]))
 
while True:
	# 输出初始界面
	print("="*12,"学员管理系统","="*14)
	print("{0:1} {1:13} {2:15}".format(" ","1. 查看学员信息","2. 添加学员信息"))
	print("{0:1} {1:13} {2:15}".format(" ","3. 删除学员信息","4. 退出系统"))
	print("="*40)

	key = input("请输入对应的选择：")
		# 根据键盘值，判断并执行对应的操作
	if key == '1':
		print("="*12,"学员信息浏览","="*14)
		ShowStu(stu)
		input("按回车键继续：")
	elif key == '2':
		print("="*12,"学员信息添加","="*14)
		name = input("请输入要添加的姓名：")
		age = int(input("请输入要添加的年龄："))
		classid = input("请输入要添加的班级名称：")
		data = (name,age,classid)
		stu.insert(data)
		input('请按回车键更新显示！')
		ShowStu(stu)
		input("按回车键继续：")
	elif key == '3':
		print("="*12,"学员信息删除","="*14)
		ShowStu(stu)
		sid = int(input("请输入你要删除的学员id号："))
		stu.delete(sid)
		input('请按回车键更新显示！')
		ShowStu(stu)
		input("按回车键继续：")
	elif key == '4':
		print("="*12,"再见","="*14)
		break
	else:
		print("======== 无效的键盘输入！ ==========")

